feat(envs): add RoboCerebra long-horizon manipulation benchmark#3314
Open
pkooij wants to merge 2 commits intofeat/benchmark-cifrom
Open
feat(envs): add RoboCerebra long-horizon manipulation benchmark#3314pkooij wants to merge 2 commits intofeat/benchmark-cifrom
pkooij wants to merge 2 commits intofeat/benchmark-cifrom
Conversation
865c2a1 to
cc8c571
Compare
e89e6d9 to
927118e
Compare
Introduces a new `robocerebra` environment type that wraps the LIBERO libero_10 suite with RoboCerebra-specific defaults (256×256 resolution, 20 FPS, camera keys `image`/`wrist_image` matching the HF dataset). Also fixes a gap in the LIBERO factory chain: `camera_name_mapping` was accepted by `LiberoEnv.__init__` but never forwarded through `create_libero_envs` → `_make_env_fns` → `_make_env`. Both `LiberoEnv` and `RoboCerebraEnv` now propagate it correctly end-to-end. - `envs/configs.py`: `RoboCerebraEnv` config + `LiberoEnv.create_envs` fix - `envs/libero.py`: thread `camera_name_mapping` through factory chain - `pyproject.toml`: `robocerebra` optional dep group (= `lerobot[libero]`) - `tests/envs/test_robocerebra_env.py`: 10 unit tests (no LIBERO needed) - `docs/source/robocerebra.md`: install, dataset, eval commands, citation Dataset: CollisionCode/RoboCerebra_lerobot_v3.0 (already LeRobot v3 — no conversion needed) Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
Adds the RoboCerebra benchmark to the benchmark integration test suite introduced in #3309. Follows the same pattern: one isolated Docker image per benchmark so dependency trees cannot collide. - docker/Dockerfile.benchmark.robocerebra: installs lerobot[robocerebra] only (= lerobot[libero] alias: hf-libero + dm-control + mujoco) - .github/workflows/benchmark_tests.yml: full workflow with libero, metaworld, and robocerebra parallel jobs; robocerebra job builds its own image and runs a 1-episode smoke eval on libero_10 Note: benchmark_tests.yml is also created in #3309. Whichever PR merges second will need a trivial conflict resolution (add the robocerebra job block to the existing file). Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
cc8c571 to
0a39e51
Compare
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
robocerebraenv type wrapping LIBERO'slibero_10suite with RoboCerebra-specific defaults (256×256, 20 FPS, camera keysimage/wrist_imagematching the HF dataset)camera_name_mappingwas accepted byLiberoEnv.__init__but never forwarded throughcreate_libero_envs → _make_env_fns → _make_env; now correctly propagated end-to-end (also benefits existingLiberoEnvusers who pass--env.camera_name_mappingvia CLI)robocerebraoptional dependency group inpyproject.toml(aliaseslerobot[libero])Dataset
The dataset
CollisionCode/RoboCerebra_lerobot_v3.0is already in LeRobot v3.0 format — no conversion needed.Camera keys in the dataset match our defaults:
observation.images.image(agent-view)observation.images.wrist_image(wrist)Files changed
src/lerobot/envs/configs.pyRoboCerebraEnv; fixLiberoEnv.create_envsto forwardcamera_name_mappingsrc/lerobot/envs/libero.pycamera_name_mappingthroughcreate_libero_envs → _make_env_fns → _make_envpyproject.tomlrobocerebraextra (= lerobot[libero], Linux only)tests/envs/test_robocerebra_env.pydocs/source/robocerebra.mdTest plan
pytest tests/envs/test_robocerebra_env.py) — no LIBERO needed, all env creation mockedpre-commit run -apasses on all changed filesPublishing proposal
The dataset is already on HuggingFace at
CollisionCode/RoboCerebra_lerobot_v3.0. To mirror it under thelerobot/org:Requires
lerobot/org write access — can be done separately after merge.🤖 Generated with Claude Code